LLM 分类器的电路化组合与成本优化
🧭 LLM 多分类器自适应组合与成本优化
Harvard School of Engineering and Applied Science · Kempner Institute for the Study of Natural & Artificial Intelligence · School of Electrical and Computer Engineering, Georgia Institute of Technology · Department of Electrical and Computer Engineering, University of California, Santa Barbara · Center for Computational Mathematics, Flatiron Institute
将多个 LLM 分类器组织为自适应查询电路,以更低成本突破路由/级联的准确率前沿
📌 工作参考价值(4 点)
- 在电商/广告/搜索中,LLM 常用于意图分类、商品属性打标、内容审核、相关性评分等;借鉴电路思想,将不同 prompt/模型/温度配置视为带成本的分类器,基于混淆矩阵动态调度,比固定级联更省调用成本。
- 工程实现上,可把每个 LLM 分类器封装为统一 Query/Decision 节点,离线用验证集估计各节点的 per-class 错误率和成本,再用动态规划搜索给定预算下的最优执行路径;在线按查表决策,适合批量打标和实时流量。
- 需要低成本高可靠性的 LLM 评审或 guardrail 场景,可先用便宜模型筛简单样本,对不确定性样本再追加强模型或 self-consistency,但要预先校准各分类器的条件错误率,避免误差传播。
- 对 Agent 评估 / 自动评测 / 推荐理由质量把关等任务,论文结论支持在多个弱监督信号之间做电路化融合,不等同于简单投票或级联,能更高效利用算力预算。
ReplaySSM:缓存 SSM 输入而非状态
🧭 LLM 推理加速 · SSM 状态缓存优化
Princeton University · NVIDIA
通过 checkpoint 加近期输入缓冲替代逐步状态读写,显著降低 SSM 状态 I/O 并支持高效投机解码与混合精度
📌 工作参考价值(4 点)
- 在线推理中若使用 Mamba-2、Gated DeltaNet 等 SSM 或混合模型,可参考 ReplaySSM 的 checkpoint + input buffer 设计,将状态更新从逐步改为周期性批量更新,减少约一半状态内存流量;直接降低首 token 延迟和长序列 decoding 成本,对搜索推荐场景中的 Agent 长链推理和工具调用尤其有用。
- 投机解码的 rollback 可以从复制/恢复完整状态改为只移动输入 buffer 的指针,极大降低回滚开销;在推荐系统里做批量候选生成或 LLM 推理时,可为投机验证阶段节省内存和计算,提高吞吐。
- 混合精度技巧:输出计算使用低精度 checkpoint,循环更新使用高精度 checkpoint,两个副本分离,避免量化误差进入后续 recurrent state。如果在线模型需要对状态做量化压缩,可采用此方法降低 KL 散度和下游效果损失。
- 工程实现上可以直接集成到 vLLM 等现有推理框架,不要求替换模型结构;业务中若已使用 vLLM 部署 SSM 或混合架构,可评估开启 ReplaySSM 获取 1.2-1.47x 标准解码加速或 1.8x 于投机解码的吞吐提升。
表格基础模型的注意力FP8量化
🧭 表格基础模型推理优化 · FP8注意力量化
Prior Labs
开发FP8量化QKV的注意力Triton内核,通过训练/测试量化误差对齐实现1.7倍加速且精度无损
📌 工作参考价值(4 点)
- 如果业务里用了类似 TabPFN 的 tabular transformer 做点击率、LTV、风控等表格数据建模,推理优化重点应放在 attention 计算上,而不是像 LLM 那样优先做 weight 或 KV cache 量化;QKV 直接量化到 FP8 并用显式 FP8 matmul,可以在注意力热点上拿到 1.5-1.7x 加速。
- 关键实现 trick:训练和推理时对行(row)的量化误差必须对齐,否则精度会急剧下降。迁移到推荐/广告场景时,如果表格模型在训练和线上服务的数据分布或预处理不同,需要把量化校准纳入训练管线,保证一致性。
- 动态开启量化内核的策略值得借鉴:在训练行数超过 8192 时才启用 FP8 attention kernel,短序列时用基线 kernel,避免小 batch/短序列下量化开销反而拉低吞吐。
- 工程上用 Triton kernel 可以针对 attention 做轻量级量化优化,无需重新训练模型;在表格特征建模或 in-context tabular learning 的推理服务中,可以将 FP8 attention 作为无精度损失的加速开关,降低线上延迟和成本。
DeaMoE:面向小批量快速解码的高效 MoE 结构
🧭 高效 MoE 解码架构 · 小批量推理 ★ 深度解读
University of Science and Technology of China · Institute of Artificial Intelligence, Hefei Comprehensive National Science Center · Shanxi University
DeaMoE 将专家分组为部门,共享大部分参数并设计两级路由,在小批量解码下降低专家权重加载,实现最高 2 倍加速
📌 工作参考价值(4 点)
- 若业务用 DeepSeek-V3/Qwen3 等大 MoE 做实时 query 改写、广告文案生成或 Agent tool call,小批量延迟不达标时,可考虑 DeaMoE 式重组:把 256/384 个 experts 分成 8/16 个 departments,共享 FFN 主矩阵,每个 expert 只保留 [hffn, hffn] 私有矩阵;在总参数和 FLOPs 不变的前提下,A40 上可获得 1.3–2.0× TPOT 加速。
- 改造时用 group-limited top-k 软约束控制每个 token 激活的 department 数,不要强制覆盖 kdept 个 department;ablation 显示 strict coverage 会扰动路由偏好并提高 loss,软约束下平均激活 3.95/4 个 department 已足够。
- 私有 expert 矩阵用 torch.nn.init.eye 做 identity 初始化,而不是随机初始化;在 factorized 结构下可显著降低预训练 loss(2.504 vs 2.530),避免随机初始化扭曲共享 backbone 表示。
- 工程实现可沿用 vLLM Grouped GEMM primitives,只改 token indexing 支持 hierarchical routing;注意收益与 expert 大小和 GPU 带宽强相关,H100 上小 expert 模型可能回退,优先用于大 expert MoE 和带宽受限硬件。
DiffusionGemma:基于 Gemma 4 微调的离散扩散高速文本生成模型
🧭 文本扩散模型 · 高速块生成 ★ 深度解读
Google DeepMind
通过微调 AR 模型结合 RL 与采样器蒸馏,实现离散扩散文本生成,速度达 ~1500 TPS,远超自回归模型
📌 工作参考价值(4 点)
- 块级并行生成降低延迟:在需要实时响应的对话或文案生成场景中,可直接借鉴 DiffusionGemma 的块扩散范式,采用 256 token 并行去噪的方式代替逐 token 解码,大幅降低单用户服务延迟,尤其适合低并发、低批量的推理场景。
- RL 与采样器蒸馏联合训练可压缩推理步数:对生成式推荐或搜索改写等系统,可设计类似的在线训练流程(SD·RL),将奖励最大化与推理步数压缩作为联合目标,使模型在保持生成质量的同时大幅减少前向次数,提升吞吐。
- 自适应停止机制动态节省算力:对于不同复杂度的任务(如简单商品描述 vs. 复杂推荐理由),可以引入基于预测熵的自适应停止策略,对简单任务早期停止去噪,避免固定步数浪费计算,实现“按需计算”。
- 保留 AR 能力实现混合解码:在 Agent 系统中,可根据延迟要求和任务复杂度动态切换扩散模式或 AR 模式,或混合使用两者,平衡生成速度与质量,为请求路由策略提供新的自由度。
窗口化 MTP:消除百万级上下文中的全量草稿 KV 开销
🧭 推测解码 · 长上下文 KV 优化
NVIDIA
对推测解码草稿头施加窗口注意力,训练免费、无损地将长上下文草稿 KV 读取量降低 99%,推理加速 28–44%
📌 工作参考价值(4 点)
- 在长上下文生成式推荐/Agent 中降低延迟:当使用 LLM 生成推荐理由或进行多步推理时,若启用 MTP 推测解码,可直接将草稿头的注意力替换为窗口注意力,保留验证阶段全注意力,实现训练免费、无损的推理加速,尤其适合百万 token 级别的用户行为序列。
- 工程实现简单,即插即用:无需重新训练或调整模型结构,仅需在草稿注意力中添加滑动窗口和注意力槽(StreamingLLM 风格),可立即部署到现有 SGLang 等框架。
- 释放 GPU 显存用于更大批处理:窗口化草稿头可丢弃约 99% 的草稿 KV 缓存(实测占总量 7.7–11%),使用紧凑环形缓冲区回收,不影响接受准确率,支持更高的吞吐。
- 适用于混合注意力目标模型:当主模型使用混合线性注意力(如 Mamba2-hybrid)时,窗口化草稿头可避免其全注意力读取代价成为瓶颈,保持整体加速效果随上下文变长而放大。
FVAttn:视频生成的自适应稀疏注意力与运行时负载均衡
🧭 分布式推理·稀疏注意力负载均衡
中山大学 · Tencent Inc. (WeChat HPC) · Tencent Inc. (WeChat Vision) · 北京大学
提出训练无关的自适应稀疏注意力系统,通过运行时头迁移与空闲填充解决多GPU序列并行的负载不均,推理加速超2倍
📌 工作参考价值(4 点)
- 自适应稀疏中的 Top-p + Top-k 地板 组合策略:确保每个注意力头至少保留 k 个高相关 token,避免完全稀疏导致的训练/推理不稳定,在 LLM 长上下文推荐对话或 Agent 记忆检索时可借鉴此安全机制。
- 运行时负载均衡通过 P2P 迁移少量“重头”到轻载 GPU,这对多 GPU 部署推荐模型(如生成式推荐、多模态理解)时的序列并行头负载不均问题有直接参考价值,可降低栅极延迟。
- Slack-Aware Sparse Augmentation 利用非关键路径 GPU 的空闲时间计算额外高价值块,这种“空闲填充”思路可迁移到分布式 Agent 流水线调度,在等待 I/O 或其他 Agent 结果时执行预取或候选评分。
- 将稀疏计算与负载均衡、通信重叠结合的整体系统设计,展示了在工业级推理服务中如何兼顾精度与吞吐,推荐系统从业者可将类似思想用于大规模模型的多路召回或多专家(MoE)推理的负载均衡。
电商搜索转化率预测模型的规模化实践
🧭 搜索转化率预测 · 模型缩放与推理优化 ★ 深度解读
Coupang
系统研究搜索CVR模型在骨干、嵌入、数据三个维度的独立可加和缩放规律,并实现线上 +2.6% 转化提升
📌 工作参考价值(5 点)
- 优先选用 MaskNet 配合交叉宽度缩放:相比 DCNv2 和序列模型,MaskNet 在相同 FLOPs 下具有更高的特征交叉效率,且内存占用更友好,适合在搜索 CVR 这类强调显式高阶交互的任务中构建 Pareto-optimal 骨干。
- 独立缩放实验可加速迭代:骨干、嵌入维度和训练数据量对离线 mAP 的提升效应接近线性可加,因此可以分别在子数据集上独立调参(如先在小数据上做架构搜索),再将组合方案直接迁移到全量数据训练,大幅降低探索成本。
- 数据缩放遵循对数线性,新特征不必全量回填:训练时长增加带来的收益递减明显,且近期 300 天行为序列就足够支撑个性化嵌入效果,历史特征缺失用 “unseen” 占位不会明显损伤指标,从而可以只对高价值特征做近期回填。
- 推理优化让大模型延迟持平:采用 CPU 特征预处理与 GPU 骨干推理分离、客户端预分组 + 服务端动态批处理的混合方案,可在 8 倍推理 FLOPs 下将 P99 延迟从 82ms 降至 32ms,峰值吞吐提升 4.4 倍,避免因延迟约束放弃大模型。
- 热启动训练策略显著缩短迭代周期:利用固定维度隐层作为输入投影,使得特征增减时只需重新初始化输入映射层,其余参数全部 warm-start,可将训练时间从冷启动的 +264% 压缩到仅 +36%,适合高频模型更新。